package com.ai.demo.entity.vo;

import lombok.Data;
import lombok.NoArgsConstructor;
import org.springframework.ai.chat.messages.Message;
@NoArgsConstructor
@Data
public class MessageVo {
    private String role;
    private String content;
    public MessageVo(Message message) {
        switch (message.getMessageType()) {
            case USER:
                role = "user";
                break;
            case ASSISTANT:
                role = "assistant";
                break;
            case SYSTEM:
                role = "system";
                break;
            default:
                throw new IllegalArgumentException("Invalid message type: " + message.getMessageType());
        }
       this.content = message.getText();
    }
}

